package algorithm.sorty.sort.compare;

import algorithm.sorty.Sort;

/**
 * @Description: 归并排序
 * @Author Ammar
 * @Create 2023/5/2 19:14
 */
public class MergeSort<T extends Comparable<T>> extends Sort<T> {
    private T[] leftArray;

    @Override
    public void sort() {
        // 临时存放merge时的左边数组
        leftArray = (T[]) new Comparable[array.length >> 1];
        sort(0, array.length);
    }

    private void sort(int begin, int end) {
        if (end - begin < 2) return;

        int mid = (begin + end) >> 1;
        sort(begin, mid);
        sort(mid, end);
        merge(begin, mid, end);
    }

    // 将[begin, mid)和[mid, end)合并成一个有序序列
    private void merge(int begin, int mid, int end) {
        // li: 左边数组的起始索引
        // le: 左边数组的结束索引
        // ri: 右边数组的起始索引
        // re: 右边数组的结束索引
        // ai: merge后数组的起始索引
        int li = 0, le = mid - begin;
        int ri = mid, re = end;
        int ai = begin;

        // 备份左边数组
        for (int i = li; i < le; i++) {
            leftArray[i] = array[begin + i];
        }

        // 如果左边还没有结束
        while (li < le) {
            // 如果右边还没有结束
            if (ri < re && compare(array[ri], leftArray[li]) < 0) {
                array[ai++] = array[ri++];
            } else {
                array[ai++] = leftArray[li++];
            }
        }
    }
}
